package com.musemun.config.Listener;

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.musemun.common.ResultCode;
import com.musemun.exception.ApiException;
import com.musemun.pojo.Crifo;
import com.musemun.pojo.ExcelCrifo;
import com.musemun.pojo.OutExcel;
import com.musemun.service.CrifoService;
import com.musemun.service.function.LendContact;
import com.musemun.service.function.functionimpl.LendContactImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;

public class OutBatchListener extends AnalysisEventListener<OutExcel> {

    /**
     * 每隔5条存储数据库，实际使用中可以3000条，然后清理list ，方便内存回收
     */
    private static final int BATCH_COUNT = 5;
    List<OutExcel> list = new ArrayList<>();
    /**
     * 假设这个是一个DAO，当然有业务逻辑这个也可以是一个service。当然如果不用存储这个对象没用。
     */

    public OutBatchListener(LendContact lendContact) {
        this.lendContact = lendContact;
    }

    LendContact lendContact;





    public OutBatchListener() {
        // 这里是demo，所以随便new一个。实际使用如果到了spring,请使用下面的有参构造函数
    }

    /**
     * 如果使用了spring,请使用这个构造方法。每次创建Listener的时候需要把spring管理的类传进来
     *
     * @param demoDAO
     */
//    public ExcelListener(DemoDAO demoDAO) {
//        this.demoDAO = demoDAO;
//    }

    /**
     * 这个每一条数据解析都会来调用
     *
     * @param data
     *            one row value. Is is same as {@link AnalysisContext#readRowHolder()}
     * @param context
     */
    @Override
    @Transactional
    public void invoke(OutExcel data, AnalysisContext context) {
        System.out.println(data);
        if (data.check()) {
            list.add(data);
        }

        // 达到BATCH_COUNT了，需要去存储一次数据库，防止数据几万条数据在内存，容易OOM
        if (list.size() >= BATCH_COUNT) {
            for (OutExcel element : list) {
                System.out.println(element.getCrlendifo());
                System.out.println(element.getCropifo());
                System.out.println(element.getLendcontact());
                lendContact.insert(element.getCrlendifo(), element.getLendcontact(), element.getCropifo());

            }
            list.clear();
        }
    }


    /**
     * 所有数据解析完成了 都会来调用
     *
     * @param context
     */
    @Override
    @Transactional
    public void doAfterAllAnalysed(AnalysisContext context) {
        // 这里也要保存数据，确保最后遗留的数据也存储到数据库
        for (OutExcel element : list) {
            System.out.println(element.getCrlendifo());
            System.out.println(element.getCropifo());
            System.out.println(element.getLendcontact());
            lendContact.insert(element.getCrlendifo(), element.getLendcontact(), element.getCropifo());
        }
    }

    /**
     * 加上存储数据库
     */


}
